package com.hx.nine.eleven.web.ftp.jsch;

import com.jcraft.jsch.Channel;
import com.jcraft.jsch.ChannelSftp;
import com.jcraft.jsch.JSch;
import com.jcraft.jsch.Session;
import com.jcraft.jsch.SftpATTRS;
import com.jcraft.jsch.SftpException;
import com.jcraft.jsch.SftpProgressMonitor;
import com.jcraft.jsch.SftpStatVFS;
import com.jcraft.jsch.UIKeyboardInteractive;
import com.jcraft.jsch.UserInfo;

import javax.swing.*;
import java.awt.*;

/**
 * @auth wml
 * @date 2024/10/21
 */
public class Sftp {

	//http://www.jcraft.com/jsch/examples/Sftp.java
	public static void main(String[] arg){

		try{
			JSch jsch=new JSch();

			String host=null;
			if(arg.length>0){
				host=arg[0];
			}
			else{
				host= JOptionPane.showInputDialog("Enter username@hostname",
						System.getProperty("user.name")+
								"@localhost");
			}
			String user=host.substring(0, host.indexOf('@'));
			host=host.substring(host.indexOf('@')+1);
			int port=22;

			Session session=jsch.getSession(user, host, port);

			// username and password will be given via UserInfo interface.
			UserInfo ui=new MyUserInfo();
			session.setUserInfo(ui);

			session.connect();

			Channel channel=session.openChannel("sftp");
			channel.connect();
			ChannelSftp c=(ChannelSftp)channel;

			java.io.InputStream in=System.in;
			java.io.PrintStream out=System.out;

			java.util.Vector cmds=new java.util.Vector();
			byte[] buf=new byte[1024];
			int i;
			String str;
			int level=0;

			while(true){
				out.print("sftp> ");
				cmds.removeAllElements();
				i=in.read(buf, 0, 1024);
				if(i<=0)break;

				i--;
				if(i>0 && buf[i-1]==0x0d)i--;
				//str=new String(buf, 0, i);
				//System.out.println("|"+str+"|");
				int s=0;
				for(int ii=0; ii<i; ii++){
					if(buf[ii]==' '){
						if(ii-s>0){ cmds.addElement(new String(buf, s, ii-s)); }
						while(ii<i){if(buf[ii]!=' ')break; ii++;}
						s=ii;
					}
				}
				if(s<i){ cmds.addElement(new String(buf, s, i-s)); }
				if(cmds.size()==0)continue;

				String cmd=(String)cmds.elementAt(0);
				if(cmd.equals("quit")){
					c.quit();
					break;
				}
				if(cmd.equals("exit")){
					c.exit();
					break;
				}
				if(cmd.equals("rekey")){
					session.rekey();
					continue;
				}
				if(cmd.equals("compression")){
					if(cmds.size()<2){
						out.println("compression level: "+level);
						continue;
					}
					try{
						level=Integer.parseInt((String)cmds.elementAt(1));
						if(level==0){
							session.setConfig("compression.s2c", "none");
							session.setConfig("compression.c2s", "none");
						}
						else{
							session.setConfig("compression.s2c", "zlib@openssh.com,zlib,none");
							session.setConfig("compression.c2s", "zlib@openssh.com,zlib,none");
						}
					}
					catch(Exception e){}
					session.rekey();
					continue;
				}
				if(cmd.equals("cd") || cmd.equals("lcd")){
					if(cmds.size()<2) continue;
					String path=(String)cmds.elementAt(1);
					try{
						if(cmd.equals("cd")) c.cd(path);
						else c.lcd(path);
					}
					catch(SftpException e){
						System.out.println(e.toString());
					}
					continue;
				}
				if(cmd.equals("rm") || cmd.equals("rmdir") || cmd.equals("mkdir")){
					if(cmds.size()<2) continue;
					String path=(String)cmds.elementAt(1);
					try{
						if(cmd.equals("rm")) c.rm(path);
						else if(cmd.equals("rmdir")) c.rmdir(path);
						else c.mkdir(path);
					}
					catch(SftpException e){
						System.out.println(e.toString());
					}
					continue;
				}
				if(cmd.equals("chgrp") || cmd.equals("chown") || cmd.equals("chmod")){
					if(cmds.size()!=3) continue;
					String path=(String)cmds.elementAt(2);
					int foo=0;
					if(cmd.equals("chmod")){
						byte[] bar=((String)cmds.elementAt(1)).getBytes();
						int k;
						for(int j=0; j<bar.length; j++){
							k=bar[j];
							if(k<'0'||k>'7'){foo=-1; break;}
							foo<<=3;
							foo|=(k-'0');
						}
						if(foo==-1)continue;
					}
					else{
						try{foo=Integer.parseInt((String)cmds.elementAt(1));}
						catch(Exception e){continue;}
					}
					try{
						if(cmd.equals("chgrp")){ c.chgrp(foo, path); }
						else if(cmd.equals("chown")){ c.chown(foo, path); }
						else if(cmd.equals("chmod")){ c.chmod(foo, path); }
					}
					catch(SftpException e){
						System.out.println(e.toString());
					}
					continue;
				}
				if(cmd.equals("pwd") || cmd.equals("lpwd")){
					str=(cmd.equals("pwd")?"Remote":"Local");
					str+=" working directory: ";
					if(cmd.equals("pwd")) str+=c.pwd();
					else str+=c.lpwd();
					out.println(str);
					continue;
				}
				if(cmd.equals("ls") || cmd.equals("dir")){
					String path=".";
					if(cmds.size()==2) path=(String)cmds.elementAt(1);
					try{
						java.util.Vector vv=c.ls(path);
						if(vv!=null){
							for(int ii=0; ii<vv.size(); ii++){
//		out.println(vv.elementAt(ii).toString());

								Object obj=vv.elementAt(ii);
								if(obj instanceof com.jcraft.jsch.ChannelSftp.LsEntry){
									out.println(((com.jcraft.jsch.ChannelSftp.LsEntry)obj).getLongname());
								}

							}
						}
					}
					catch(SftpException e){
						System.out.println(e.toString());
					}
					continue;
				}
				if(cmd.equals("lls") || cmd.equals("ldir")){
					String path=".";
					if(cmds.size()==2) path=(String)cmds.elementAt(1);
					try{
						java.io.File file=new java.io.File(path);
						if(!file.exists()){
							out.println(path+": No such file or directory");
							continue;
						}
						if(file.isDirectory()){
							String[] list=file.list();
							for(int ii=0; ii<list.length; ii++){
								out.println(list[ii]);
							}
							continue;
						}
						out.println(path);
					}
					catch(Exception e){
						System.out.println(e);
					}
					continue;
				}
				if(cmd.equals("get") ||
						cmd.equals("get-resume") || cmd.equals("get-append") ||
						cmd.equals("put") ||
						cmd.equals("put-resume") || cmd.equals("put-append")
				){
					if(cmds.size()!=2 && cmds.size()!=3) continue;
					String p1=(String)cmds.elementAt(1);
//	  String p2=p1;
					String p2=".";
					if(cmds.size()==3)p2=(String)cmds.elementAt(2);
					try{
						SftpProgressMonitor monitor=new MyProgressMonitor();
						if(cmd.startsWith("get")){
							int mode=ChannelSftp.OVERWRITE;
							if(cmd.equals("get-resume")){ mode=ChannelSftp.RESUME; }
							else if(cmd.equals("get-append")){ mode=ChannelSftp.APPEND; }
							c.get(p1, p2, monitor, mode);
						}
						else{
							int mode=ChannelSftp.OVERWRITE;
							if(cmd.equals("put-resume")){ mode=ChannelSftp.RESUME; }
							else if(cmd.equals("put-append")){ mode=ChannelSftp.APPEND; }
							c.put(p1, p2, monitor, mode);
						}
					}
					catch(SftpException e){
						System.out.println(e.toString());
					}
					continue;
				}
				if(cmd.equals("ln") || cmd.equals("symlink") ||
						cmd.equals("rename") || cmd.equals("hardlink")){
					if(cmds.size()!=3) continue;
					String p1=(String)cmds.elementAt(1);
					String p2=(String)cmds.elementAt(2);
					try{
						if(cmd.equals("hardlink")){  c.hardlink(p1, p2); }
						else if(cmd.equals("rename")) c.rename(p1, p2);
						else c.symlink(p1, p2);
					}
					catch(SftpException e){
						System.out.println(e.toString());
					}
					continue;
				}
				if(cmd.equals("df")){
					if(cmds.size()>2) continue;
					String p1 = cmds.size()==1 ? ".": (String)cmds.elementAt(1);
					SftpStatVFS stat = c.statVFS(p1);

					long size = stat.getSize();
					long used = stat.getUsed();
					long avail = stat.getAvailForNonRoot();
					long root_avail = stat.getAvail();
					long capacity = stat.getCapacity();

					System.out.println("Size: "+size);
					System.out.println("Used: "+used);
					System.out.println("Avail: "+avail);
					System.out.println("(root): "+root_avail);
					System.out.println("%Capacity: "+capacity);

					continue;
				}
				if(cmd.equals("stat") || cmd.equals("lstat")){
					if(cmds.size()!=2) continue;
					String p1=(String)cmds.elementAt(1);
					SftpATTRS attrs=null;
					try{
						if(cmd.equals("stat")) attrs=c.stat(p1);
						else attrs=c.lstat(p1);
					}
					catch(SftpException e){
						System.out.println(e.toString());
					}
					if(attrs!=null){
						out.println(attrs);
					}
					else{
					}
					continue;
				}
				if(cmd.equals("readlink")){
					if(cmds.size()!=2) continue;
					String p1=(String)cmds.elementAt(1);
					String filename=null;
					try{
						filename=c.readlink(p1);
						out.println(filename);
					}
					catch(SftpException e){
						System.out.println(e.toString());
					}
					continue;
				}
				if(cmd.equals("realpath")){
					if(cmds.size()!=2) continue;
					String p1=(String)cmds.elementAt(1);
					String filename=null;
					try{
						filename=c.realpath(p1);
						out.println(filename);
					}
					catch(SftpException e){
						System.out.println(e.toString());
					}
					continue;
				}
				if(cmd.equals("version")){
					out.println("SFTP protocol version "+c.version());
					continue;
				}
				if(cmd.equals("help") || cmd.equals("?")){
					out.println(help);
					continue;
				}
				out.println("unimplemented command: "+cmd);
			}
			session.disconnect();
		}
		catch(Exception e){
			System.out.println(e);
		}
		System.exit(0);
	}

	public static class MyUserInfo implements UserInfo, UIKeyboardInteractive {
		public String getPassword(){ return passwd; }
		public boolean promptYesNo(String str){
			Object[] options={ "yes", "no" };
			int foo=JOptionPane.showOptionDialog(null,
					str,
					"Warning",
					JOptionPane.DEFAULT_OPTION,
					JOptionPane.WARNING_MESSAGE,
					null, options, options[0]);
			return foo==0;
		}

		String passwd;
		JTextField passwordField=(JTextField)new JPasswordField(20);

		public String getPassphrase(){ return null; }
		public boolean promptPassphrase(String message){ return true; }
		public boolean promptPassword(String message){
			Object[] ob={passwordField};
			int result=
					JOptionPane.showConfirmDialog(null, ob, message,
							JOptionPane.OK_CANCEL_OPTION);
			if(result==JOptionPane.OK_OPTION){
				passwd=passwordField.getText();
				return true;
			}
			else{ return false; }
		}
		public void showMessage(String message){
			JOptionPane.showMessageDialog(null, message);
		}
		final GridBagConstraints gbc =
				new GridBagConstraints(0,0,1,1,1,1,
						GridBagConstraints.NORTHWEST,
						GridBagConstraints.NONE,
						new Insets(0,0,0,0),0,0);
		private Container panel;
		public String[] promptKeyboardInteractive(String destination,
												  String name,
												  String instruction,
												  String[] prompt,
												  boolean[] echo){
			panel = new JPanel();
			panel.setLayout(new GridBagLayout());

			gbc.weightx = 1.0;
			gbc.gridwidth = GridBagConstraints.REMAINDER;
			gbc.gridx = 0;
			panel.add(new JLabel(instruction), gbc);
			gbc.gridy++;

			gbc.gridwidth = GridBagConstraints.RELATIVE;

			JTextField[] texts=new JTextField[prompt.length];
			for(int i=0; i<prompt.length; i++){
				gbc.fill = GridBagConstraints.NONE;
				gbc.gridx = 0;
				gbc.weightx = 1;
				panel.add(new JLabel(prompt[i]),gbc);

				gbc.gridx = 1;
				gbc.fill = GridBagConstraints.HORIZONTAL;
				gbc.weighty = 1;
				if(echo[i]){
					texts[i]=new JTextField(20);
				}
				else{
					texts[i]=new JPasswordField(20);
				}
				panel.add(texts[i], gbc);
				gbc.gridy++;
			}

			if(JOptionPane.showConfirmDialog(null, panel,
					destination+": "+name,
					JOptionPane.OK_CANCEL_OPTION,
					JOptionPane.QUESTION_MESSAGE)
					==JOptionPane.OK_OPTION){
				String[] response=new String[prompt.length];
				for(int i=0; i<prompt.length; i++){
					response[i]=texts[i].getText();
				}
				return response;
			}
			else{
				return null;  // cancel
			}
		}
	}

/*
  public static class MyProgressMonitor implements com.jcraft.jsch.ProgressMonitor{
    JProgressBar progressBar;
    JFrame frame;
    long count=0;
    long max=0;
    public void init(String info, long max){
      this.max=max;
      if(frame==null){
        frame=new JFrame();
	frame.setSize(200, 20);
        progressBar = new JProgressBar();
      }
      count=0;
      frame.setTitle(info);
      progressBar.setMaximum((int)max);
      progressBar.setMinimum((int)0);
      progressBar.setValue((int)count);
      progressBar.setStringPainted(true);
      JPanel p=new JPanel();
      p.add(progressBar);
      frame.getContentPane().add(progressBar);
      frame.setVisible(true);
      System.out.println("!info:"+info+", max="+max+" "+progressBar);
    }
    public void count(long count){
      this.count+=count;
      System.out.println("count: "+count);
      progressBar.setValue((int)this.count);
    }
    public void end(){
      System.out.println("end");
      progressBar.setValue((int)this.max);
      frame.setVisible(false);
    }
  }
*/

	public static class MyProgressMonitor implements SftpProgressMonitor{
		ProgressMonitor monitor;
		long count=0;
		long max=0;
		public void init(int op, String src, String dest, long max){
			this.max=max;
			monitor=new ProgressMonitor(null,
					((op==SftpProgressMonitor.PUT)?
							"put" : "get")+": "+src,
					"",  0, (int)max);
			count=0;
			percent=-1;
			monitor.setProgress((int)this.count);
			monitor.setMillisToDecideToPopup(1000);
		}
		private long percent=-1;
		public boolean count(long count){
			this.count+=count;

			if(percent>=this.count*100/max){ return true; }
			percent=this.count*100/max;

			monitor.setNote("Completed "+this.count+"("+percent+"%) out of "+max+".");
			monitor.setProgress((int)this.count);

			return !(monitor.isCanceled());
		}
		public void end(){
			monitor.close();
		}
	}

	private static String help =
			"      Available commands:\n"+
					"      * means unimplemented command.\n"+
					"cd path                       Change remote directory to 'path'\n"+
					"lcd path                      Change local directory to 'path'\n"+
					"chgrp grp path                Change group of file 'path' to 'grp'\n"+
					"chmod mode path               Change permissions of file 'path' to 'mode'\n"+
					"chown own path                Change owner of file 'path' to 'own'\n"+
					"df [path]                     Display statistics for current directory or\n"+
					"                              filesystem containing 'path'\n"+
					"help                          Display this help text\n"+
					"get remote-path [local-path]  Download file\n"+
					"get-resume remote-path [local-path]  Resume to download file.\n"+
					"get-append remote-path [local-path]  Append remote file to local file\n"+
					"hardlink oldpath newpath      Hardlink remote file\n"+
					"*lls [ls-options [path]]      Display local directory listing\n"+
					"ln oldpath newpath            Symlink remote file\n"+
					"*lmkdir path                  Create local directory\n"+
					"lpwd                          Print local working directory\n"+
					"ls [path]                     Display remote directory listing\n"+
					"*lumask umask                 Set local umask to 'umask'\n"+
					"mkdir path                    Create remote directory\n"+
					"put local-path [remote-path]  Upload file\n"+
					"put-resume local-path [remote-path]  Resume to upload file\n"+
					"put-append local-path [remote-path]  Append local file to remote file.\n"+
					"pwd                           Display remote working directory\n"+
					"stat path                     Display info about path\n"+
					"exit                          Quit sftp\n"+
					"quit                          Quit sftp\n"+
					"rename oldpath newpath        Rename remote file\n"+
					"rmdir path                    Remove remote directory\n"+
					"rm path                       Delete remote file\n"+
					"symlink oldpath newpath       Symlink remote file\n"+
					"readlink path                 Check the target of a symbolic link\n"+
					"realpath path                 Canonicalize the path\n"+
					"rekey                         Key re-exchanging\n"+
					"compression level             Packet compression will be enabled\n"+
					"version                       Show SFTP version\n"+
					"?                             Synonym for help";
}
